

package com.hk.common.utils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import java.util.List;

/**
 * IP地址
 *
 * @author zhangy
 */
public class IPUtils {

    private static Logger logger = LoggerFactory.getLogger(IPUtils.class);

    /**
     * 获取IP地址
     * <p>
     * 使用Nginx等反向代理软件， 则不能通过request.getRemoteAddr()获取IP地址
     * 如果使用了多级反向代理的话，X-Forwarded-For的值并不止一个，而是一串IP地址，X-Forwarded-For中第一个非unknown的有效IP字符串，则为真实IP地址
     */
    public static String getIpAddr(HttpServletRequest request) {
        String ip = null;
        try {
            ip = request.getHeader("x-forwarded-for");
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("Proxy-Client-IP");
            }
            if (StringUtils.isEmpty(ip) || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("WL-Proxy-Client-IP");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("HTTP_CLIENT_IP");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("HTTP_X_FORWARDED_FOR");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getRemoteAddr();
            }
        } catch (Exception e) {
            logger.error("IPUtils ERROR ", e);
        }

        return ip;
    }



    /**
     * 校验IP是否属于白名单
     * @param ip
     * @return
     */
    public static boolean checkIP(String ip,List<String> ipList){
        // 如果是本地ip直接允许
        if("0:0:0:0:0:0:0:1".equals(ip)){
            return true;
        }

        // 循环判断
        for (String str: ipList ) {
            if(!StringUtils.isEmpty(str)){
                if(!str.contains("-")){
                    // 不包含-的单独ip地址匹配上则返回，否则不处理，进行下一次匹配
                    if(str.trim().equals(ip.trim())){
                        return true;
                    }
                }else{
                    String[] strArr = str.split("-");
                    // 如果长度不为2 则设定不合法，继续下一次匹配
                    if(strArr.length!=2){
                        continue;
                    }
                    // 只有成功才返回，否则不处理，进行下一次匹配
                    if(ipInRange(strArr[0],strArr[1],ip)){
                        return true;
                    }
                }
            }
        }
        return false;
    }

    /**
     * 判断ip是否在范围内，必须是同网段
     * @param begin
     * @param end
     * @param ip
     * @return
     */
    private static boolean ipInRange(String begin, String end, String ip){
        // 任意为空则返回false
        if(StringUtils.isEmpty(begin)|| StringUtils.isEmpty(end)){
            return false;
        }
        try{
            String beginPrefix = begin.substring(0,begin.lastIndexOf("."));
            String endPrefix = end.substring(0,end.lastIndexOf("."));
            String ipPrefix = ip.substring(0,ip.lastIndexOf("."));
            if(beginPrefix.equals(endPrefix)&&beginPrefix.equals(ipPrefix)){
                String beginLast = begin.substring(begin.lastIndexOf(".")+1);
                String endLast = end.substring(end.lastIndexOf(".")+1);
                String ipLast = ip.substring(ip.lastIndexOf(".")+1);
                // 只有前值相同，后值在范围内
                if(Integer.parseInt(beginLast)<=Integer.parseInt(ipLast) && Integer.parseInt(endLast)>=Integer.parseInt(ipLast)){
                    return true;
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }
        return false;
    }

}
